package org.suze.demo.route;

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

/**
 * 自定义 AbstractRoutingDataSource 实现类
 */
public class DynamicRoutingDataSource extends AbstractRoutingDataSource {

    /**
     * AbstractRoutingDataSource 通过这个方法，选择配置的数据源中的一个使用，返回配置数据源的键
     * @return
     */
    @Override
    protected DataSourceEnum determineCurrentLookupKey() {
        // 获取当前线程使用数据源
        return DynamicDataSourceHandler.get();
    }

}